#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
AI Agent 快速开始示例
演示如何使用 AI Agent 执行简单的任务
"""
from colorama import init, Fore, Style

from agent import AIAgent

# 初始化colorama
init()

def quick_demo():
    """快速开始示例"""
    print(f"{Fore.CYAN} AI Agent 快速开始演示{Style.RESET_ALL}")
    print("=" * 50)

    # 初始化 Agent
    print("1. 初始化AI Agent...")
    agent = AIAgent("openai")
    print()

    # 示例任务
    demo_tasks = [
        "创建一个名为 'demo.txt' 的文件，内容为 'Hello from AI Agent!'",
        "计算 10 + 20 * 3 的结果",
        "获取当前目录的文件列表"
    ]

    for i, task in enumerate(demo_tasks, 1):
        print(f"{Fore.YELLOW}任务 {i}: {task}{Style.RESET_ALL}")
        print("-" * 40)

        try:
            result = agent.execute_task(task)
            print(f"{Fore.GREEN} √ 任务{i} 完成{Style.RESET_ALL}")
            print(f"成功步骤：{result['successful_steps']}")
            print(f"失败步骤：{result['failed_steps']}")
            print()
        except Exception as e:
            print(f"{Fore.RED} × 任务 {i} 失败：{str(e)}{Style.RESET_ALL}")
            print()

    print(f"{Fore.MAGENTA} 演示完成！{Style.RESET_ALL}")



if __name__ == "__main__":
    quick_demo()

